for dataset in ['METR-LA', 'PEMS-BAY']:
    for ratio in [0.25, 0.5, 0.75]:
        print('python main.py --dataset {dataset}-{ratio} --seed 42 --model_name SplitFedAvgNodePredictor --base_model_name GRUSeq2SeqWithGraphNet --batch_size 128 --server_batch_size 48 --hidden_size 64 --suffix mp --use_curriculum_learning --mp_worker_num 8 --sync_every_n_epoch 1 --server_epoch 20 --gcn_on_server --gpus 0, --max_epochs 200 --early_stop_patience 20 --gru_num_layers 1'.format(dataset=dataset, ratio=ratio))
        print('python main.py --dataset {dataset}-{ratio} --seed 42 --model_name FedAvgNodePredictor --base_model_name GRUSeq2Seq --batch_size 128 --hidden_size 200 --gru_num_layers 2 --use_curriculum_learning --mp_worker_num 8 --sync_every_n_epoch 1 --cl_decay_steps 1 --max_epochs 200 --early_stop_patience 20'.format(dataset=dataset, ratio=ratio))
